package com.example.leetcode.c101_c200.c129;
/**
 * 给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
 *
 * 例如，从根到叶子节点路径 1->2->3 代表数字 123。
 *
 * 计算从根到叶子节点生成的所有数字之和。
 *
 * 说明: 叶子节点是指没有子节点的节点。
 *
 * 示例 1:
 *
 * 输入: [1,2,3]
 *     1
 *    / \
 *   2   3
 * 输出: 25
 * 解释:
 * 从根到叶子节点路径 1->2 代表数字 12.
 * 从根到叶子节点路径 1->3 代表数字 13.
 * 因此，数字总和 = 12 + 13 = 25.
 * 示例 2:
 *
 * 输入: [4,9,0,5,1]
 *     4
 *    / \
 *   9   0
 *  / \
 * 5   1
 * 输出: 1026
 * 解释:
 * 从根到叶子节点路径 4->9->5 代表数字 495.
 * 从根到叶子节点路径 4->9->1 代表数字 491.
 * 从根到叶子节点路径 4->0 代表数字 40.
 * 因此，数字总和 = 495 + 491 + 40 = 1026.
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */

import com.example.leetcode.common.TreeNode;
import com.example.leetcode.common.TreeNodeWrapper;

import java.util.ArrayList;
import java.util.List;

/**
 * 求根到叶子节点数字之和
 * @author zhanpengguo
 * @date 2020-10-29 8:19
 */
public class Solution {

    private List<StringBuffer> list = new ArrayList<>();
    private int sum = 0;

    public int sumNumbers(TreeNode root) {
        if (root == null) {
            return 0;
        }
        dfs(new StringBuffer(), root);
        return list.stream().mapToInt(e->Integer.parseInt(e.toString())).sum();
    }

    private void dfs(StringBuffer sb, TreeNode root) {
        sb.append(root.val);
        if (root.left == null && root.right == null) {
            list.add(new StringBuffer(sb));
            sb.deleteCharAt(sb.length() - 1);
            return;
        }
        if (root.left != null) {
            dfs(sb, root.left);
        }
        if (root.right != null) {
            dfs(sb, root.right);
        }
        sb.deleteCharAt(sb.length() - 1);
    }

    public int sumNumbers1(TreeNode root) {
        return dfs1(root, 0);
    }

    private int dfs1(TreeNode root, int prevSum) {
        if (root == null) {
            return 0;
        }
        int sum = prevSum * 10 + root.val;
        if (root.left == null && root.right == null) {
            return sum;
        } else {
            return dfs1(root.left, sum) + dfs1(root.right, sum);
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        TreeNode root = TreeNodeWrapper.stringToTreeNode("[4,9,0,5,1]");
        int i = solution.sumNumbers1(root);
        System.out.println(solution.list);
        System.out.println(i);
    }
}
